/*
 *
 *  * Copyright (C) 2015 Square, Inc.
 *  *
 *  * Licensed under the Apache License, Version 2.0 (the "License");
 *  * you may not use this file except in compliance with the License.
 *  * You may obtain a copy of the License at
 *  *
 *  *      http://www.apache.org/licenses/LICENSE-2.0
 *  *
 *  * Unless required by applicable law or agreed to in writing, software
 *  * distributed under the License is distributed on an "AS IS" BASIS,
 *  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  * See the License for the specific language governing permissions and
 *  * limitations under the License.
 *
 */
package com.lark.oapi.okhttp.internal.connection;

import com.lark.oapi.okhttp.Route;

import java.util.LinkedHashSet;
import java.util.Set;

/**
 * A blacklist of failed routes to avoid when creating a new connection to a target address. This is
 * used so that OkHttp can learn from its mistakes: if there was a failure attempting to connect to
 * a specific IP address or proxy server, that failure is remembered and alternate routes are
 * preferred.
 */
final class RouteDatabase {

    private final Set<Route> failedRoutes = new LinkedHashSet<>();

    /**
     * Records a failure connecting to {@code failedRoute}.
     */
    public synchronized void failed(Route failedRoute) {
        failedRoutes.add(failedRoute);
    }

    /**
     * Records success connecting to {@code route}.
     */
    public synchronized void connected(Route route) {
        failedRoutes.remove(route);
    }

    /**
     * Returns true if {@code route} has failed recently and should be avoided.
     */
    public synchronized boolean shouldPostpone(Route route) {
        return failedRoutes.contains(route);
    }
}
